from graphrag.chat_friend import ChatFriend

if __name__ == "__main__":
    chat_friend = ChatFriend()
    while True:
        user_input = input("用户：").strip()
        if user_input in ["退出", "quit", "exit"]:
            chat_friend.save_graph()
            break
        if not user_input:
            continue
        
        chat_friend.invoke(user_input)

        